Insertion sort in Python